Skip to content

Ci rocm setup fix #238

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 1 commit into
base: rocm-main
Choose a base branch
from
Open

Ci rocm setup fix #238

wants to merge 1 commit into from

Conversation

Ruturaj4
Copy link

@Ruturaj4 Ruturaj4 commented Feb 25, 2025

Updated dependency constraint for jax-rocm60-plugin: Previously, it was dynamically set based on _current_jaxlib_version and _jax_version. It is now fixed to jax-rocm60-plugin>=0.5.0 to ensure compatibility with the intended JAX and ROCm stack.

@Ruturaj4 Ruturaj4 force-pushed the ci_rocm_setup_fix branch 2 times, most recently from bffeb5d to b3a81f6 Compare February 26, 2025 12:18
@Ruturaj4 Ruturaj4 requested a review from a team as a code owner February 26, 2025 12:18
@charleshofer
Copy link
Collaborator

Rebased this to see if CI passes

@Ruturaj4 Ruturaj4 added the open-upstream Tag when you want a copy of this PR to be opened on upstream label Mar 10, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
open-upstream Tag when you want a copy of this PR to be opened on upstream
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants